package leetcode.doublepointer;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;

// 剑指 Offer II 074. 合并区间
public class Merge {

    public static void main(String[] args) {
        int[][] data = {{1, 9}, {2, 5}, {19, 20}, {10, 11}, {12, 20}, {0, 3}, {0, 1}, {0, 2}};
        int[][] ints = merge(data);
        for (int i = 0; i < ints.length; i++) {
            for (int j = 0; j < ints[0].length; j++) {
                System.out.print("\t" + ints[i][j]);
            }
            System.out.println();
        }
    }

    /**
     * 我们用数组 merged 存储最终的答案。
     * <p>
     * 首先，我们将列表中的区间按照左端点升序排序。然后我们将第一个区间加入 merged 数组中，
     * 并按顺序依次考虑之后的每个区间：
     * <p>
     * 如果当前区间的左端点在数组 merged 中最后一个区间的右端点之后，那么它们不会重合，
     * 我们可以直接将这个区间加入数组 merged 的末尾；
     * <p>
     * 否则，它们重合，我们需要用当前区间的右端点更新数组 merged 中最后一个区间的右端点，
     * 将其置为二者的较大值。
     *
     * @param intervals
     * @return
     */
    public static int[][] merge(int[][] intervals) {
        if (intervals.length == 0) {
            return new int[0][2];
        }
        Arrays.sort(intervals, (a, b) -> a[0] - b[0]);
        List<int[]> merged = new ArrayList<int[]>();
        for (int i = 0; i < intervals.length; ++i) {
            int L = intervals[i][0], R = intervals[i][1];
            if (merged.size() == 0 || merged.get(merged.size() - 1)[1] < L) {
                merged.add(new int[]{L, R});
            } else {
                merged.get(merged.size() - 1)[1] = Math.max(merged.get(merged.size() - 1)[1], R);
            }
        }
        return merged.toArray(new int[merged.size()][]);

    }
}
